make animations in python